import torch

# 检测能否使用GPU
print(torch.cuda.is_available())

# 随机生成一个2*3的二维数组
x = torch.rand(2, 3)
print("x = ", x)

# 随机生成一个2*3的二维数组
y = torch.rand(2, 3)
print("y = ", y)

# 矩阵求和
z0 = torch.add(x, y)    # 方法1 方法计算，赋给新值
z1 = x + y              # 方法2 运算符计算，赋给新值
x.add_(y)               # 方法3 一个值的方法，加上另一个值（改写第一个值）
print("z0 = ", z0)
print("z1 = ", z1)
print("x = ", x)

# 获取指定列
print("x[:, 0] = ", x[:, 0])

x = x * 3

# 获取指定列
print("x[:, 0] = ", x[:, 0])

#